from selenium.webdriver.common.by import By
from page_object.pages.BasePage import BasePage
from page_object.pages.MyPage import MyPage
from page_object.pages.SearchPage import SearchPage
from page_object.pages.SelectedPage import SelectedPage


class MainPage(BasePage):
    _search_button = (By.ID, "home_search")
    """
    首页
    """
    def gotoSelected(self):
        """
        点击进入行情页面
        :return: 返回行情页面
        """
        # 第一种写法，也是最基础得写法
        # self.driver.find_element_by_xpath("//*[@text='行情']")
        # self.driver.find_element_by_xpath("//*[@text='行情']").click()

        # 第二种写法，by写法
        # by写法，BasePage里面得find方法，写法：（by.定位方式, 参数）
        # hangqing = (By.XPATH, "//*[@text='行情']")
        # self.find(hangqing)
        # self.find(hangqing).click()

        # 第三种写法，调用BasePage里面封装得find_by_xpath方法，最简单
        self.find_by_xpath("行情")
        self.find_by_xpath("行情").click()

        return SelectedPage()

    def gotoSearch(self) -> SearchPage:
        self.find(self._search_button).click()
        return SearchPage()
    def gotoMyPage(self):
        My_button = (By.XPATH, "//*[contains(@resource-id, 'tab_name') and (@text = '我的')]/..")
        self.find(My_button).click()
        return MyPage()
